import time
from wordpress_project.base.base_page import Tools


class new_tool(Tools):
    login_url = "http://192.168.0.125:9009/wp-login.php"  # 网址
    username_ele = '[id="user_login"]'  # 用户名css定位
    password_ele = '[name="pwd"]'  # 密码css定位
    login_button_ele = '[name="wp-submit"]'  # 登录按钮css定位
    username = '[id = "user_login"]'  # 根据css定位到用户名输入框
    email = '[id="email"]'  # 根据css定位到电子邮件输入框
    button_add_user = '[id="createusersub"]'  # 根据css定位到添加用户按钮
    add_user_cg = '[id="message"] p'  # 新增成功定位
    add_user_sb = '[class="error"]'  # 注册失败定位
    add_user_fh = '[class="page-title-action"]'  # 新建成功，返回

    def new_action(self, d_usr, pwd):
        self.visit(self.login_url)
        # 在方法中，调用同一个类的其他方法   self.方法名  在方法中，调用类的变量   self.变量名
        self.by_css(self.username_ele).send_keys(d_usr)
        time.sleep(1)
        self.by_css(self.password_ele).send_keys(pwd)
        time.sleep(1)
        self.by_css(self.login_button_ele).click()

    def new_add_action(self, user, eml):
        self.by_css(self.username).send_keys(user)
        time.sleep(1)
        self.by_css(self.email).send_keys(eml)
        time.sleep(1)
        self.by_css(self.button_add_user).click()
        time.sleep(1)

    def get_cg_msg(self):
        return self.by_css(self.add_user_cg).text

    def get_sb_msg(self):
        return self.by_css(self.add_user_sb).text
